#encoding=utf-8
from my_bonito.utils.get_model_instance import get_model_instance
import my_bonito.config as config


instance = get_model_instance(config.MODEL)

def get_raw_outputs(prompt, path):
    return instance.get_raw_output(prompt, path)


if __name__ == '__main__':
    prompt_example = '''
    请结合文档内容和你自己的知识以及互联网知识，分别提出question answering without choices、extractive question answering、yes-no question answering、question answering without choices、extractive question answering、yes-no question answering、natural language inference、multiple-choice question answering等不同任务类别的指令任务，指令任务不能显现有文档的存在，即不能出现"根据文档"等短语或文档的标题，之后给出问题回答。格式为:<TAKSTYPE>given type</TASKTYPE><QUESTION>你根据文档内容提出的问题</QUESTIONS><ANSWER>你的回答</ANSWER>:
    '''
    path_example = r'S:\Code\my_self_instruct\my_bonito\data\source_files\mind.docx'
    raw_outputs = get_raw_outputs(prompt_example, path_example)
    print(raw_outputs)